#ifndef MATH_NMS_BOX_H
#define MATH_NMS_BOX_H

#include <cuda_runtime.h>
#include <vector>
#include "../rt_mem_fly.hpp"
#include "cuda_common.hpp"
namespace algocommon
{

void NMSBoxTensorStream(TensorFloat &sorted_box_tensor,
                        TensorFloat &nms_box_tensor,
                        float nms_thres,
                        cudaStream_t stream, int topk = 500);

} // namespace algocommon

#endif